import torch
import numpy as np
import easyocr
# 检查版本
print(f"PyTorch 版本: {torch.__version__}")  # 2.2.2+cu121
print(f"NumPy 版本: {np.__version__}")      #1.26.4

# 检查设备
print(f"CUDA 可用: {torch.cuda.is_available()}")
print(f"当前设备: {torch.device('cuda' if torch.cuda.is_available() else 'cpu')}")

# 初始化 EasyOCR
reader = easyocr.Reader(['en'], gpu=torch.cuda.is_available())
print(f"EasyOCR 使用设备: {reader.device}")  # 应为 "cuda:0" 或 "cpu"

# 测试识别
result = reader.readtext(np.ones((100, 300, 3), dtype=np.uint8) * 255)
print("测试成功！")